package com.ce.server.service.impl;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.OSSException;
import com.aliyun.oss.model.CannedAccessControlList;
import com.aliyun.oss.model.ObjectMetadata;
import com.ce.server.common.OssEntity;
import com.ce.server.service.IAliOssService;
import org.joda.time.DateTime;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.attribute.FileTime;
import java.util.Date;
import java.util.UUID;

/**
 * @DATE: 2021/11/21 10:55
 * @Author: 小爽帅到拖网速
 */
@Service
public class AliOssServiceImpl implements IAliOssService, InitializingBean {

  @Autowired
  private OssEntity ossEntity;
  /**
   * yourEndpoint填写Bucket所在地域对应的Endpoint。以华东1（杭州）为例，
   * Endpoint填写为https://oss-cn-hangzhou.aliyuncs.com。
   */
  private String endpoint;
  /**
   * 阿里云账号AccessKey拥有所有API的访问权限，风险很高。
   * 强烈建议您创建并使用RAM用户进行API访问或日常运维，请登录RAM控制台创建RAM用户
   */
  private String accessKeyId;
  private String accessKeySecret;
  /**
   * 填写Bucket名称，例如examplebucket。
   */
  private String bucketName;

  /**
   * 初始化bean之后要进行的操作
   *
   * @throws Exception
   */
  @Override
  public void afterPropertiesSet() throws Exception {
    endpoint = ossEntity.getEndpoint();
    accessKeyId = ossEntity.getAccessKeyId();
    accessKeySecret = ossEntity.getAccessKeySecret();
    bucketName = ossEntity.getBucketName();
  }

  /**
   * 创建存储空间
   */
  @Override
  public void createBucket() {
    OSS ossClient = null;
    try {
      // 创建OSSClient实例。
      ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
      // 判断bucket是否存在
      if (ossClient.doesBucketExist(bucketName)) {
        throw new RuntimeException(bucketName + "在对象存储的Bucket列表中已经存在");
      }
      // 创建存储空间。
      ossClient.createBucket(bucketName);
    } catch (OSSException e) {
      e.printStackTrace();
    } finally {
      // 关闭OSSClient。
      ossClient.shutdown();
    }
  }

  /**
   * 上传文件
   *
   * @param file
   * @return
   */
  @Override
  public String upload(MultipartFile file) {
    // 上传地址
    String uploadUrl = null;

    OSS ossClient = null;
    try {
      // 创建OSSClient实例。
      ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
      // 判断bucketName是否存在
      if (!ossClient.doesBucketExist(bucketName)) {
        // 创建bucket
        ossClient.setBucketAcl(bucketName, CannedAccessControlList.PublicRead);
      }
      //获取上传文件输入流
      InputStream inputStream = file.getInputStream();
      //获取上传文件的全名称
      String originalFilename = file.getOriginalFilename();

      // 在文件名称里面添加随机唯一的值
      String uuid = UUID.randomUUID().toString().replaceAll("-", "");

      // 获取上传文件的拓展名
      String fileType = originalFilename.substring(originalFilename.lastIndexOf("."));

      // 把文件按照日期进行分类
      //获取当前日期
      //2021/11/21
      String datePath = new DateTime().toString("yyyy/MM/dd");
      //拼接
      // 拼接文件名称 uuid.png
      String newFileName = uuid + fileType;

      // 生成文件夹
      // 2021/11/21/uuid.png
      String finalFileName = datePath + "/" + newFileName;

      // 如果要实现图片预览的效果，一定要设置以下几点
      // 1、设置文件的ACL权限，要么是公共读，要么是公共读写
      // 2、一定要设置文本类型（image/jpg）
      ObjectMetadata objectMetadata = new ObjectMetadata();
      // 设置公共读权限
      objectMetadata.setObjectAcl(CannedAccessControlList.PublicRead);
      objectMetadata.setContentType(getcontentType(fileType));


      //调用oss方法实现上传
      //第一个参数  Bucket名称
      //第二个参数  上传到oss文件路径和文件名称   aa/bb/1.jpg
      //第三个参数  上传文件输入流
      ossClient.putObject(bucketName, finalFileName, inputStream,objectMetadata);

      // 默认十年不过期
      Date expiration = new Date(System.currentTimeMillis() + 3600L * 1000 * 24 * 365 * 10);
      // bucket名称 文件名 过期时间
      uploadUrl = ossClient.generatePresignedUrl(bucketName,finalFileName,expiration).toString();

    } catch (Exception e) {
      e.printStackTrace();
    } finally {
      // 关闭OSSClient。
      ossClient.shutdown();
    }
    return uploadUrl.substring(0,uploadUrl.indexOf("?"));
    // return uploadUrl;
  }

  @Override
  public void download(String fileName) throws IOException {

  }

  @Override
  public void listFile() {

  }

  @Override
  public void deleteFile(String filename) {

  }

  /**
   * Description: 判断OSS服务文件上传时文件的contentType
   *
   * @param FilenameExtension 文件后缀
   * @return String
   */
  public static String getcontentType(String filenameExtension) {
    if (filenameExtension.equalsIgnoreCase("bmp")) {
      return "image/bmp";
    }
    if (filenameExtension.equalsIgnoreCase("gif")) {
      return "image/gif";
    }
    if (filenameExtension.equalsIgnoreCase("jpeg") || filenameExtension.equalsIgnoreCase("jpg")
            || filenameExtension.equalsIgnoreCase("png")) {
      return "image/jpeg";
    }
    if (filenameExtension.equalsIgnoreCase("html")) {
      return "text/html";
    }
    if (filenameExtension.equalsIgnoreCase("txt")) {
      return "text/plain";
    }
    if (filenameExtension.equalsIgnoreCase("vsd")) {
      return "application/vnd.visio";
    }
    if (filenameExtension.equalsIgnoreCase("pptx") || filenameExtension.equalsIgnoreCase("ppt")) {
      return "application/vnd.ms-powerpoint";
    }
    if (filenameExtension.equalsIgnoreCase("docx") || filenameExtension.equalsIgnoreCase("doc")) {
      return "application/msword";
    }
    if (filenameExtension.equalsIgnoreCase("xml")) {
      return "text/xml";
    }
    return "image/jpeg";
  }

}
